###
 # @Author: xushaocong
 # @Date: 2022-08-24 13:17:03
 # @LastEditTime: 2022-08-24 13:22:03
 # @LastEditors: xushaocong
 # @Description: 
 # @FilePath: /Group-Free-3D/my_script/train.sh
 # email: xushaocong@stu.xmu.edu.cn
### 

port_num=29530

DATA_ROOT=datasets/
gpu_ids="2,3"
gpu_num=2
# b_size=10
b_size=8

python -m torch.distributed.launch --master_port $port_num --nproc_per_node $gpu_num \
    train_dist.py --num_point 50000 --num_decoder_layers 6 \
    --size_delta 0.111111111111 --center_delta 0.04 \
    --learning_rate 0.006 --decoder_learning_rate 0.0006 --weight_decay 0.0005 \
    --dataset scannet --data_root datasets --log_dir logs


